
package com.shop.cloud.mall.admin.controller;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.alibaba.nacos.api.config.ConfigService;
import com.alibaba.nacos.api.exception.NacosException;
import com.shop.cloud.common.core.constant.CommonConstants;
import com.shop.cloud.common.core.util.EnvUtils;
import com.shop.cloud.common.core.util.R;
import com.shop.cloud.common.core.util.SensitiveUtils;
import com.shop.cloud.common.im.config.IMConfigProperties;
import com.shop.cloud.common.log.annotation.SysLog;
import com.shop.cloud.common.nacos.util.CommentUtils;
import com.shop.cloud.common.nacos.util.NacosConfigUtils;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Operation;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
import org.yaml.snakeyaml.Yaml;

import java.util.HashMap;
import java.util.Map;

/**
 * IM配置
 *
 * @author
 * @date 2020-11-24 14:44:15
 */
@Slf4j
@RestController
@AllArgsConstructor
@RequestMapping("/imconfig")
@Tag(description = "imconfig", name = "IM配置管理")
public class IMConfigController {

	private NacosConfigUtils nacosConfigUtils;

    /**
     * im配置查询
     * @param
     * @return R
     */
    @Operation(summary = "im配置查询")
    @GetMapping
    @PreAuthorize("@ato.hasAuthority('mall:imconfig:get')")
    public R getBy() throws NacosException {
		ConfigService configService = nacosConfigUtils.getConfigService();
		String content = configService.getConfig(EnvUtils.getConfigDataId(CommonConstants.CONFIG_DATA_ID_APPLICATION), CommonConstants.CONFIG_GROUP, CommonConstants.CONFIG_TIMEOUT_MS);
		Yaml yaml = new Yaml();
		Map<String, Object> confMap= yaml.load(content);
		Object object = confMap.get("im");
		JSONObject jSONObject = JSONUtil.parseObj(object);
		IMConfigProperties imConfigProperties = jSONObject.toBean(IMConfigProperties.class);
		//数据脱敏
		String sensitiveSecret = SensitiveUtils.process(imConfigProperties.getSecret());
		imConfigProperties.setSecret(sensitiveSecret);
		return R.ok(imConfigProperties);
    }

    /**
     * im配置修改
     * @param imConfigProperties im配置
     * @return R
     */
    @Operation(summary = "im配置修改")
    @SysLog("修改im配置")
    @PutMapping
    @PreAuthorize("@ato.hasAuthority('mall:imconfig:edit')")
    public R update(@RequestBody IMConfigProperties imConfigProperties) throws NacosException {
		ConfigService configService = nacosConfigUtils.getConfigService();
		String content = configService.getConfig(EnvUtils.getConfigDataId(CommonConstants.CONFIG_DATA_ID_APPLICATION), CommonConstants.CONFIG_GROUP, CommonConstants.CONFIG_TIMEOUT_MS);
		// 记录 yaml 文件的注释信息
		CommentUtils.CommentHolder holder = CommentUtils.buildCommentHolder(content);
		Yaml yaml = new Yaml();
		Map<String, Map> confMap= yaml.load(content);

		String secret = confMap.get("im").get("secret")+"";
		String sensitiveSecret = SensitiveUtils.process(secret);
		if(!sensitiveSecret.equals(imConfigProperties.getSecret())){
			secret = imConfigProperties.getSecret();
		}

		Map imMap = new HashMap();
		imMap.put("url",imConfigProperties.getUrl());
		imMap.put("secret",secret);
		confMap.put("im",imMap);
		String confStr = yaml.dumpAsMap(confMap);
		// 填充注释信息
		confStr = holder.fillComments(confStr);
		boolean rs = configService.publishConfig(EnvUtils.getConfigDataId(CommonConstants.CONFIG_DATA_ID_APPLICATION), CommonConstants.CONFIG_GROUP,confStr,"yaml");
        return R.ok(rs);
    }

}
